set(TEST_PRIM_PURE_NEW_IR_CASES test_prim_program test_prim_simpnet
                                test_prim_custom_vjp test_prim_jit)

foreach(target ${TEST_PRIM_PURE_NEW_IR_CASES})
  py_test_modules(${target} MODULES ${target} ENVS GLOG_v=1
                  FLAGS_enable_pir_api=true)
endforeach()

file(
  GLOB TEST_PRIM_TRANS_NEW_IR_CASES
  RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
  "test_*.py")
string(REPLACE ".py" "" TEST_PRIM_TRANS_NEW_IR_CASES
               "${TEST_PRIM_TRANS_NEW_IR_CASES}")

list(REMOVE_ITEM TEST_PRIM_TRANS_NEW_IR_CASES ${TEST_PRIM_PURE_NEW_IR_CASES})

foreach(target ${TEST_PRIM_TRANS_NEW_IR_CASES})
  py_test_modules(${target} MODULES ${target} ENVS GLOG_v=1
                  FLAGS_enable_new_ir_in_executor=true)
endforeach()
